#!/usr/bin/python3

import os
from langchain.llms import OpenAI
from langchain.chains.question_answering import load_qa_chain
from langchain.docstore.document import Document

os.environ["OPENAI_API_KEY"] = ""

# 初始化 OpenAI 语言模型
llm = OpenAI(
    api_key="",
    base_url="http://zhr.icu:11434/v1",
    temperature=0.7
)

# 定义一个简单的文档
docs = [Document(page_content="苹果公司是一家美国的科技公司，由史蒂夫·乔布斯、史蒂夫·沃兹尼亚克和韦恩在 1976 年创立。苹果以其创新的电子产品如 iPhone、iPad 和 Mac 而闻名。")]

# 加载问答链
chain = load_qa_chain(llm, chain_type="stuff")

# 提出问题
question = "苹果公司是哪一年创立的？"

# 运行问答链获取答案
answer = chain.run(input_documents=docs, question=question)

print(f"问题: {question}")
print(f"答案: {answer}")